//给你两个数组，arr1 和 arr2，arr2 中的元素各不相同，arr2 中的每个元素都出现在 arr1 中。 
//
// 对 arr1 中的元素进行排序，使 arr1 中项的相对顺序和 arr2 中的相对顺序相同。未在 arr2 中出现过的元素需要按照升序放在 arr1 的末
//尾。 
//
// 
//
// 示例 1： 
//
// 
//输入：arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
//输出：[2,2,2,1,4,3,3,9,6,7,19]
// 
//
// 示例 2: 
//
// 
//输入：arr1 = [28,6,22,8,44,17], arr2 = [22,28,8,6]
//输出：[22,28,8,6,17,44]
// 
//
// 
//
// 提示： 
//
// 
// 1 <= arr1.length, arr2.length <= 1000 
// 0 <= arr1[i], arr2[i] <= 1000 
// arr2 中的元素 arr2[i] 各不相同 
// arr2 中的每个元素 arr2[i] 都出现在 arr1 中 
// 
//
// Related Topics 数组 哈希表 计数排序 排序 👍 303 👎 0


package LeetCode.editor.cn;

import java.util.*;

/**
 * @author ldltd
 * @date 2025-03-02 00:09:49
 * @description 1122.数组的相对排序
 */
public class RelativeSortArray{
	 public static void main(String[] args) {
	 	 //测试代码
	 	 RelativeSortArray fun=new RelativeSortArray();
	 	 Solution solution = fun.new Solution();

	 }
	 
//力扣代码
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
		 //自定义排序
	public int[] relativeSortArray(int[] arr1, int[] arr2) {
		Map<Integer, Integer> map = new HashMap<>();
		List<Integer> list = new ArrayList<>();
		for(int num : arr1) list.add(num);
		for(int i = 0; i < arr2.length; i++) map.put(arr2[i], i);
		Collections.sort(list, (x, y) -> {
			if(map.containsKey(x) || map.containsKey(y)) return map.getOrDefault(x, 1001) - map.getOrDefault(y, 1001);
			return x - y;
		});
		for(int i = 0; i < arr1.length; i++) arr1[i] = list.get(i);
		return arr1;
	}
	//计数排序
	public int[] relativeSortArray1(int[] arr1, int[] arr2) {
		int upper = 0;
		//找出ar1最大值
		for (int x : arr1) {
			upper = Math.max(upper, x);
		}
		//计数
		int[] frequency = new int[upper + 1];
		for (int x : arr1) {
			++frequency[x];
		}
		//答案
		int[] ans = new int[arr1.length];
		int index = 0;
		//遍历ar2中的每个数
		for (int x : arr2) {
			//根据计数个数放入答案中
			for (int i = 0; i < frequency[x]; ++i) {
				ans[index++] = x;
			}
			frequency[x] = 0;
		}
		//把还存在ar1中没有放入的，按顺序放入
		for (int x = 0; x <= upper; ++x) {
			for (int i = 0; i < frequency[x]; ++i) {
				ans[index++] = x;
			}
		}
		return ans;
	}

}
//leetcode submit region end(Prohibit modification and deletion)

}
